# 定义要处理的文件列表
files=(
  "ceval.jsonl.t"
  "FLUB.jsonl.t"
  "gsm8k_main.jsonl.t"
  "ifeval_input_data.jsonl.t"
  "logitQA.jsonl.t"
  "math23k_test.json.t"
  "math401.json.t"
  "mmlu.jsonl.t"
)

gpu_id=3
input_dir="tgt_t2t"
epoch="tgt_t2t_34_47499"
checkpoint_path="/mnt/sfs/asr/code/osum_xlgeng/examples/wenetspeech/whisper/exp/epoch_34_LLMinstruct_cosyvoice1_10Wtts_1WenTTS_2Khqtts_1KenS2S_3Ks2s_5Ws2t/step_47499.pt"
config_path="/mnt/sfs/asr/code/osum_xlgeng/examples/wenetspeech/whisper/conf/config_llm_huawei_instruct-version_cosyvoice1-token.yaml"

# 创建输出目录
mkdir -p "$epoch"

# 遍历处理每个文件
for file in "${files[@]}"; do
  input_jsonl_path="../$input_dir/$file"
  output_jsonl_path="../$epoch/$file"
  
  echo "正在处理文件: $input_jsonl_path"
  python3 infer_text2text.py $gpu_id $checkpoint_path $config_path $input_jsonl_path $output_jsonl_path
  echo "已完成处理，输出到: $output_jsonl_path"
done